package com.iceland.common.message;

import java.util.HashMap;
import java.util.Map;

public class Header {
    Map<Integer,String> map=new HashMap<>();

//    public Integer getCode(){
//        if(!map.containsKey(HeaderType.CODE)){
//            return null;
//        }
//        return Integer.valueOf(get(HeaderType.CODE));
//    }
//    public String getMessage(){
//        if(!map.containsKey(HeaderType.MESSAGE)){
//            return null;
//        }
//        return get(HeaderType.MESSAGE);
//    }


    public String get(int key){
        return map.get(key);
    }
    public String get(HeaderType key){
        return get(key.getType());
    }
    public void set(int key,String value){
        map.put(key,value);
    }
    public void set(int key,int value){
        set(key,String.valueOf(value));
    }
    public void set(HeaderType key,String value){
        set(key.getType(),value);
    }
    public void set(HeaderType key,int value){
        set(key.getType(),value);
    }

    public int size(){
        return map.size();
    }
}
